logic with python